from django.core.management.base import BaseCommand
from import_data.models import TrainMedicalData, ValMedicalData

class Command(BaseCommand):
    help = 'Delete all records from TrainMedicalData and ValMedicalData tables'

    def handle(self, *args, **kwargs):
        TrainMedicalData.objects.all().delete()
        ValMedicalData.objects.all().delete()
        self.stdout.write(self.style.SUCCESS('Successfully deleted all records from TrainMedicalData and ValMedicalData tables'))
